package com.cheng.example.atguigu.design_pattern.principle.ocp;

/*
 * @author 程帅
 * @createTime: 2020/10/18 20:51
 * @Created: with IntelliJ IDEA.
 */
public class Ocp {

    public static void main(String[] args) {
        //使用看着存在的问题
        GraphicEditor graphicEditor = new GraphicEditor();
        graphicEditor.drawShape(new Rectangle());
        graphicEditor.drawCircle(new Circle());
    }
}

//这是一个用于绘图的类
class GraphicEditor {

    //接收Shape对象，然后根据type,来绘制不同的图形
    public void drawShape(Shape s) {
        if (s.m_type == 1) {
            drawRectangle(s);
        } else if (s.m_type == 2) {
            drawShape(s);
        } else if (s.m_type == 3) {
            drawTriangle(s);
        }
    }

    //绘制矩形
    public void drawRectangle(Shape r) {
        System.out.println(" 绘制矩形");
    }

    //绘制圆形
    public void drawCircle(Shape r) {
        System.out.println(" 绘制圆形");
    }

    //绘制三角形
    public void drawTriangle(Shape r) {
        System.out.println(" 绘制三角形");
    }
}

//Shape类，基类
class Shape {
    int m_type;
}

//矩形
class Rectangle extends Shape {
    public Rectangle() {
        super.m_type = 1;
    }
}

//圆形
class Circle extends Shape {
    public Circle() {
        super.m_type = 2;
    }
}

//三角形
class Triangle extends Shape {
    public Triangle() {
        super.m_type = 3;
    }
}